"""
Copyright (c) 2022 Huawei Technologies Co.,Ltd.

openGauss is licensed under Mulan PSL v2.
You can use this software according to the terms and conditions of the Mulan PSL v2.
You may obtain a copy of Mulan PSL v2 at:

          http://license.coscl.org.cn/MulanPSL2

THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
See the Mulan PSL v2 for more details.
"""
"""
Case Type   : 故障&可靠性测试
Case Name   : recovery_max_workers为1时，备机备份延迟
Create At   : 2021/03/10
@zou_jialiang0505328126
Description :
    1.设置recovery_max_workers为1
    2.设置延迟为5min
    3.切换备节点为级联备
    4.创建表
    5.查询各节点
    6.插入数据30s后更新数据
    7.5/30/60s查询备节点
Expect      :
    1.设置成功
    2.设置成功
    3.切换成功
    4.建表成功
    5.1min后主备一致
    6.更新数据成功
    7.30s后备节点有插入数据，再过30s备节点数据被更新
History     :
    modified：2021/05/17  modified by @wan005: 修改延迟等待时间，规避时间差问题
    modified：2021/07/22  modified by @wan005: 优化查询方法 查询完毕再断言
    modified：2021/07/29  modified by @wan005: 先drop再设置备机延迟5min 校验结果
"""

import unittest
import time
from yat.test import Node
from yat.test import macro
from testcase.utils.Common import Common
from testcase.utils.CommonSH import CommonSH
from testcase.utils.CommonSH import RestartDbCluster
from testcase.utils.Constant import Constant
from testcase.utils.Logger import Logger


class RecoveryDelay(unittest.TestCase):
    commonshpri = CommonSH('PrimaryDbUser')
    nodes_tuple = ('PrimaryDbUser', 'Standby1DbUser', 'Standby2DbUser')

    @RestartDbCluster(*nodes_tuple)
    def setUp(self):
        self.log = Logger()
        self.log.info('----Opengauss_Reliability_Dbsys_Case261.py 开始执行------')
        self.db_primary_root_node = Node(node='PrimaryRoot')
        self.db_standby_node =  \
            [Node(node='Standby1Root'), Node(node='Standby2Root')]
        self.db_standby_user_node =  \
            [Node(node='Standby1DbUser'), Node(node='Standby2DbUser')]
        self.db_primary_user_node = Node(node='PrimaryDbUser')
        self.comshsta = \
            [CommonSH('Standby1DbUser'), CommonSH('Standby2DbUser')]
        self.constant = Constant()
        self.com_pri = Common('PrimaryRoot')
        self.com_sta = [Common('Standby1Root'), Common('Standby2Root')]
        self.tb_name = 'tb_259'

        self.log.info('=====安装ntp服务=======')
        install_cmd = "yum install -y ntp && " \
                      "systemctl enable ntpd && systemctl start ntpd"
        for i in range(2):
            result = self.db_standby_node[i].sh(install_cmd).result()
            self.log.info(result)

        self.log.info('------同步集群时间--------')
        for i in range(2):
            current = self.db_primary_root_node.sh(
                "date \"+%m/%d/%Y %H:%M:%S\"").result()
            self.log.info(current)
            datecmd = f'date -s "{current}";hwclock --systohc;' \
                f'service ntpd stop;ntpdate ntp.api.bz;service ntpd start'
            self.log.info(datecmd)
            result = self.db_standby_node[i].sh(datecmd).result()
            self.log.info(result)

    def test_recoverydelay(self):
        self.log.info('-----设置recovery_max_workers为1---')
        result = self.commonshpri.executeGsguc(
            'reload', self.constant.GSGUC_SUCCESS_MSG,
            'recovery_max_workers=1')
        self.assertTrue(result)

        self.log.info('-----------------删除表----------------')
        sql = f"drop table if exists {self.tb_name}"
        result = self.commonshpri.executDbSql(sql)
        self.log.info(result)
        self.assertIn(self.constant.TABLE_DROP_SUCCESS, result)

        self.log.info('-----设置recovery_min_apply_delay=5min----')
        result = self.commonshpri.executeGsguc(
            'reload', self.constant.GSGUC_SUCCESS_MSG,
            "recovery_min_apply_delay='5min'")
        self.assertTrue(result)

        self.log.info('-------将备节点切换为级联备---------')
        result = self.comshsta[0].buildStandby('-M cascade_standby')
        self.assertIn(self.constant.REBUILD_SUCCESS_MSG, result)
        result = self.comshsta[0].exec_refresh_conf()
        self.assertTrue(result)

        self.log.info('--------等待主备一致------------')
        result = self.comshsta[1].check_data_consistency()
        self.assertTrue(result)
        for i in range(90):
            result = self.commonshpri.check_cascade_standby_consistency()
            if result:
                break
            time.sleep(20)
        self.assertTrue(result)

        self.log.info('-----------------创建表----------------')
        sql = f"create table {self.tb_name}(i int, s char(10));"
        result = self.commonshpri.executDbSql(sql)
        self.log.info(result)
        self.assertIn(self.constant.TABLE_CREATE_SUCCESS, result)

        self.log.info('--------------查询各节点------------------')
        sql = f"select sysdate;select * from {self.tb_name};"
        result = self.commonshpri.executDbSql(sql)
        result_s1 = self.comshsta[0].executDbSql(sql)
        result_s2 = self.comshsta[1].executDbSql(sql)
        self.log.info(result)
        self.log.info(result_s1)
        self.log.info(result_s2)
        self.assertIn('(0 rows)', result)
        self.assertTrue(self.constant.NOT_EXIST in result_s1
                        and self.constant.NOT_EXIST in result_s2)
        time.sleep(310)
        for i in range(2):
            result = self.comshsta[i].executDbSql(sql)
            self.log.info(result)
            self.assertIn('(0 rows)', result)

        self.log.info('----------插入数据30s后更新数据--------------')
        sql = f"insert into {self.tb_name} values(1,'test');"
        result = self.commonshpri.executDbSql(sql)
        self.log.info(result)
        self.assertIn('INSERT', result)
        time.sleep(30)
        sql = f"update {self.tb_name} set i=5;"
        result = self.commonshpri.executDbSql(sql)
        self.log.info(result)
        self.assertIn('UPDATE', result)

        self.log.info('--------------查询各节点------------------')
        time.sleep(5)
        sql = f"select sysdate;select * from {self.tb_name};"
        result = self.commonshpri.executDbSql(sql)
        self.log.info(result)
        self.assertIn('5 | test', result)
        for i in range(2):
            result = self.comshsta[i].executDbSql(sql)
            self.log.info(result)
            self.assertIn('(0 rows)', result)
        time.sleep(290)
        for i in range(2):
            result = self.comshsta[i].executDbSql(sql)
            self.log.info(result)
            self.assertIn('1 | test', result)
        time.sleep(40)
        for i in range(2):
            result = self.comshsta[i].executDbSql(sql)
            self.log.info(result)
            self.assertIn('5 | test', result)

    def tearDown(self):
        self.log.info('-----------------------------环境清理--------------------')
        self.log.info('-------------还原参数----------')
        result = self.commonshpri.executeGsguc(
            'reload', self.constant.GSGUC_SUCCESS_MSG,
            'recovery_min_apply_delay=0')
        self.log.info(result)

        result = self.commonshpri.executeGsguc(
            'reload', self.constant.GSGUC_SUCCESS_MSG,
            'recovery_max_workers=4')
        self.log.info(result)

        self.log.info('--------还原集群-------')
        self.comshsta[0].buildStandby('-M standby')
        result = self.comshsta[0].exec_refresh_conf()
        self.log.info(result)

        self.log.info('-----------重启数据库-----------')
        result = self.commonshpri.stopDbCluster()
        self.log.info(result)
        result = self.commonshpri.startDbCluster()
        self.log.info(result)
        self.log.info('---Opengauss_Reliability_Dbsys_Case261.py 执行结束-------')